# Copyright (c) 2023, NVIDIA CORPORATION.  All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
set(CTEST_USE_LAUNCHERS 1)

find_package(Catch2 REQUIRED)
find_package(Jsoncpp REQUIRED)
find_package(fmt REQUIRED)

include(CTest)
include(Catch)

if (BUILD_TESTING)

    add_executable(dcgmitests)
    target_sources(dcgmitests
        PRIVATE
        DcgmiTestsMain.cpp
        DiagTests.cpp
        HealthTests.cpp
        MigTests.cpp
        OutputTests.cpp
        MigTestsHelper.cpp
        WildcardTests.cpp
        QueryTests.cpp
        DcgmiCommonTests.cpp
    )

    # NOTE: linking dcgmlib is not allowed for this library. Instead, make a mock implementation
    #       for any libdcgm function that is needed.    
    target_link_libraries(dcgmitests
        PRIVATE
            dcgmi_objects
            nvvs_interface
            diag_interface
            dcgmmodulediag_common
            buildinfo_objects
            sdk_nvml_essentials_objects
            sdk_nvml_interface
            Catch2::Catch2
            dcgm
            dcgm_common
            dcgm_logging
            dcgm_mutex
            ${CMAKE_THREAD_LIBS_INIT}
            ${JSONCPP_STATIC_LIBS}
            fmt::fmt
            rt
            dl
    )
    if (${LINK_INJECTION_LIB} STREQUAL "true")
        target_link_libraries(dcgmitests PRIVATE nvml_injection)
    endif()
    #target_include_directories(dcgmitests PRIVATE ${JSONCPP_INCLUDE_DIR})

    if (${CMAKE_SYSTEM_PROCESSOR} STREQUAL "x86_64")
        catch_discover_tests(dcgmitests EXTRA_ARGS --use-colour yes)
    endif()
endif()
